package com.example.mq.mqserver.core;

import com.example.mq.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 表示一个队列
 */
@Data
public class MQueue {
    // 队列的身份标识
    private String name;
    // 是否持久化
    private boolean durable =false;
    // 是否是独有的（只能被一个消费者使用）
    private boolean exclusive = false;
    // 无人使用后，是否自动删除
    private boolean autoDelete = false;
    // 创建队列时指定的参数选项
    private Map<String, Object> arguments = new HashMap<>();

    private ObjectMapper objectMapper = new ObjectMapper();

    // 表示当前队列有哪些消费者订阅了
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();
    // 用来实现轮询策略，记录当前循环的位置
    private AtomicInteger consumerSeq = new AtomicInteger(0);


    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArguments(String key, Object value) {
        arguments.put(key, value);
    }

    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }

    // 实现 arguments 序列化和反序列化，
    // 序列化为 json 字符串后，存入数据库
    // 从数据库中取出，反序列化为 HashMap
    // 重载 getter 和 setter 方法，给数据库操作调用
    public String getArguments() {
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
            // 序列化出现异常了，就返回一个空 JSON 字符串
            return "{}";
        }
    }
    public void setArguments(String argumentsJson) {
        try {
            arguments = objectMapper.readValue(argumentsJson, new TypeReference<Map<String, Object>>() {});
            // 因为要转化的类型比较复杂，所以需要搭配 TypeReference 来描述，具体的类型
            // 简单的类型，第二个参数直接写要转换的类型即可。
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }


    // 添加一个新的订阅者
    public void addConsumerEnv (ConsumerEnv consumerEnv) {
        consumerEnvList.add(consumerEnv);
    }
    // 挑选一个订阅者，用来处理当前的消息（采用轮询的方式）
    public ConsumerEnv chooseConsumerEnv () {
        if (consumerEnvList .size() == 0) {
            // 该队列没有消费者订阅
            return null;
        }
        // 计算当前要取的元素的下标
        int index = consumerSeq.get() % consumerEnvList.size();
        // 以原子方式将当前值增加 1
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }
}
